#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# @Project : chat_model 
# @File    : knowledge_schema.py
# @IDE     : PyCharm 
# @Author  :ZH
# @Time    : 2025/1/10 15:25
from typing import List, Optional

from pydantic import Field, BaseModel
from datetime import datetime

from apps.utils.enumerate import EmbeddingServerEnum


class BaseKnowledgeSchema(BaseModel):
    knowledge_name: str = Field(description='知识库名称', title='knowledge_name')
    description: str = Field(description='描述', title='description')
    embedding_server_id: int = Field(description='向量服务id', title='embedding_server_id')
    file_extract_id: int = Field(description='文件提取服务id', title='file_extract_id')
    user_id: int = Field(description='用户id', title='user_id')


class CreateKnowledgeSchema(BaseModel):
    __doc__ = "创建知识库模式"


class UpdateKnowledgeSchema(BaseModel):
    __doc__ = "更新知识库模式"
